import type { DoubaoModelId } from "@roo-code/types"
import { doubaoDefaultModelId, doubaoModels } from "@roo-code/types"
import type { ApiHandlerOptions } from "../../shared/api"
import { BaseOpenAiCompatibleProvider } from "./base-openai-compatible-provider"

export class DoubaoHandler extends BaseOpenAiCompatibleProvider<DoubaoModelId> {
	constructor(options: ApiHandlerOptions) {
		super({
			...options,
			providerName: "豆包",
			baseURL: "https://ark.cn-beijing.volces.com/api/v3/",
			apiKey: options.doubaoApiKey,
			defaultProviderModelId: doubaoDefaultModelId,
			providerModels: doubaoModels,
			defaultTemperature: 0.7,
		})
	}
}
